import RightContent from '@/components/RightContent';
import { BookOutlined, LinkOutlined } from '@ant-design/icons';
import type { Settings as LayoutSettings } from '@ant-design/pro-components';
import { PageLoading, SettingDrawer } from '@ant-design/pro-components';
import type { RunTimeLayoutConfig } from 'umi';
import { history, Link } from 'umi';
import defaultSettings from '../config/defaultSettings';
import InitialInfoComp from './components/InitialInfoComp';
import { currentUserDetail } from './utils/currentUserRole';
import Footer from '@/components/Footer';

const isDev = process.env.NODE_ENV === 'development';
const loginUrl = '/user/login';

/** 获取用户信息比较慢的时候会展示一个 loading */
export const initialStateConfig = {
  loading: <PageLoading />,
};

/**
 * @see  https://umijs.org/zh-CN/plugins/plugin-initial-state
 * */
export async function getInitialState(): Promise<{
  settings?: Partial<LayoutSettings>;
  currentRole?: API.currentRole;
  loading?: boolean;
}> {
  // 如果不是登录页面，执行
  if (history.location.pathname !== loginUrl) {
    // 获取用户身份
    const currentUser = await currentUserDetail();
    const currentRole = currentUser.roleName;
    return {
      currentRole,
      settings: defaultSettings,
    };
  }
  return {
    settings: defaultSettings,
  };
}
const noMargin = (path) => {
  return path === '/home';
};

// ProLayout 支持的api https://procomponents.ant.design/components/layout
export const layout: RunTimeLayoutConfig = ({ initialState, setInitialState }) => {
  return {
    rightContentRender: () => <RightContent />,
    disableContentMargin: false,
    waterMarkProps: {
      content: initialState?.currentUser?.name,
    },
    footerRender: () => <Footer />,
    onPageChange: () => {
      const { location } = history;
      const pathname = location.pathname;
      try {
        // @ts-ignore
        document.getElementsByClassName('ant-layout-content')[0].style.margin = noMargin(pathname)
          ? 0
          : '24px';
      } catch (e) {}
      // 如果没有登录，重定向到 login
      if (!initialState?.currentRole && location.pathname !== loginUrl) {
        history.push(loginUrl);
      }
    },
    links: isDev
      ? [
          <Link key="openapi" to="/umi/plugin/openapi" target="_blank">
            <LinkOutlined />
            <span>OpenAPI 文档</span>
          </Link>,
          <Link to="/~docs" key="docs">
            <BookOutlined />
            <span>业务组件文档</span>
          </Link>,
        ]
      : [],
    menuHeaderRender: undefined,
    // 自定义 403 页面
    // unAccessible: <div>unAccessible</div>,
    // 增加一个 loading 的状态
    childrenRender: (children, props) => {
      // if (initialState?.loading) return <PageLoading />;
      return (
        <>
          {children}
          {true && <InitialInfoComp />}
          {
            !props.location?.pathname?.includes('/login') && null

            // <SettingDrawer
            //   disableUrlParams
            //   enableDarkTheme
            //   settings={initialState?.settings}
            //   onSettingChange={(settings) => {
            //     setInitialState((preInitialState) => ({
            //       ...preInitialState,
            //       settings,
            //     }));
            //   }}
            // />
          }
        </>
      );
    },
    ...initialState?.settings,
  };
};
